/*
 *  Brno University of Technology
 *
 *  Copyright (C) 2009-2010 by Brno University of Technology and the contributors
 *
 *  Complete list of developers available at our web site:
 *
 *       http://spl.utko.feec.vutbr.cz
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU Lesser General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with this program.  If not, see http://www.gnu.org/licenses/.
 */
package cz.vutbr.feec.utko.ef.visualizer;

import java.util.HashMap;

import cz.vutbr.feec.utko.ef.gp.tree.Node;

// TODO: Auto-generated Javadoc
/**
 * The Class VisualizerRegister contain all the visualizers that can be used in
 * visualization. If the nonterminal or terminal is of such a type that is not
 * registered in this class, the {@link DummyVisualizer} is used.
 */
public class VisualizerRegister {
	
	/** The vizualizers. */
	private static HashMap<String, IVisualizer> vizualizers = new HashMap<String, IVisualizer>();

	static {
		registerVisualizer(Integer.class, new IntegerVisualizer());
	}

	/**
	 * Register visualizer.
	 * 
	 * @param class1 the class1
	 * @param vizualizer the vizualizer
	 */
	public static void registerVisualizer(Class<? extends Object> class1,
			IVisualizer vizualizer) {
		vizualizers.put(class1.getCanonicalName(), vizualizer);
	}

	/**
	 * Visualize.
	 * 
	 * @param fileName the file name
	 * @param node the node
	 */
	public static void visualize(String fileName, Node node) {
		Object o = node.getResult();
		IVisualizer v = vizualizers.get(o.getClass().getCanonicalName());
		if (v == null) {
			// throw new RuntimeException("Visualizer for class '"
			// + o.getClass().getCanonicalName() + "' from node "
			// + node.getNonterminalName()
			// + " is not defined. You should add such"
			// + " visualizer to VisualRegister if you need to "
			// + "visualize chromozome with WITH_IMAGES parameter.");
			v = new DummyVisualizer(o.getClass().getSimpleName());
		}
		v.visualize(o, fileName);
	}
}
